import random
import time


# tools 是工具收集模块，在此处自定义，后引入其它文件

def verification_code():
    """生成验证码和时间戳的字典
    格式为：{'code': 2593, 'time': 1636609677.2891963}
    """
    aa = {'code': random.randint(1000, 9999), 'time': int(time.time())}

    return aa


def verify_presence(a1):
    """检测是否存在，如果存在是否超时，当超时或不存在则生成新的验证码"""
    # 检测是否空值
    if a1.get('time'):
        a2 = a1['time']
        # 比较现在时间差异3600秒
        if int(time.time() - int(a2)) > 50:
            # print(int(time.time() - int(a2)))

            # 如果超时重新生成
            return verification_code()
        else:
            # 否则直接使用
            return a1
    else:
        # 空值返回新生成
        return verification_code()

# # t1 = verification_code()
# t1 = {'code': 6030, 'time': ''}
# t2 = verify_presence(t1)
#
# print(t2)
